Fix Python import error in VS Code